# 开始模型训练
from ultralytics import YOLO

# 加载预训练模型
yolo = YOLO('yolo11n.pt')
# 开始训练,尽量使用官方推荐的配置数值
yolo.train(
    data='data.yml', # 数据集配置文件路径
    epochs=500,      # 训练轮次
    imgsz=640,       # 输入图片尺寸, 会将训练的图片裁剪为 imgsz
    batch=32,        # 每次训练的批量,反复验证的图片数量
    device=0         # 使用 cpu 的话,填写cpu
)